package(default_visibility = ["//maga_transformer:__subpackages__", "//internal_source:__subpackages__"])

py_binary(
    name = "fake_glm",
    srcs = ["fake_glm.py"],
    deps = [
        "//maga_transformer:utils",
        "//maga_transformer:_ft_pickler",
    ]
)

py_binary(
    name = "fake_glm_v2",
    srcs = ["fake_glm_v2.py"],
    deps = [
        "//maga_transformer:utils",
        "//maga_transformer:_ft_pickler",
        ":fake_util"
    ]
)

py_library(
    name = "fake_util",
    srcs = ["fake_util.py"],
    deps = [
        "//maga_transformer:utils",
        "//maga_transformer:_ft_pickler",
    ]
)

py_library(
    name = "save_ft_module",
    srcs = ["save_ft_module.py"],
    deps = [
        "//maga_transformer:models",
    ]
)

py_library(
    name = "model_assistant_api",
    srcs = glob([
        "api/*.py",
    ]),
)

py_library(
    name = "model_assistant",
    srcs = [
           "model_assistant_server.py",
           ":model_assistant_api"
    ],
    deps = [
        "//maga_transformer:models",
    ]
)

